import json
import Data_Model

def rawMessageToMessage(rawMessage, count=0):
    """
    Метод Json
        Обрабатываем полученый запрос с ответом GigaChata и вынимаем сообщение
            :param rawMessage: Получаем сырой ответ на POST запрос
            :return: Возвращаем сообщение от чата
    """

    jsonData = Data_Model.Root.from_dict(json.loads(rawMessage))
    content = jsonData.choices[0].message.content
    # Выводим ответ от GigaChat"""
    if count == 0:
        return content
    elif count == 1:
        return jsonData
    else:
        return "Неизвестная ошибка"